package xyz.jcat.sat.gateway.auth;

import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.reactor.filter.SaReactorFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import xyz.jcat.common.web.Resp;
import xyz.jcat.common.web.RespCodeMsg;

/**
 * [Sa-Token 权限认证] 配置类
 */
@Slf4j
@Configuration
public class SaTokenConfigure {
    // 注册 Sa-Token全局过滤器
    @Bean
    public SaReactorFilter getSaReactorFilter() {
        return new SaReactorFilter()
                // 拦截地址
                .addInclude("/**")
                // 开放地址
                .addExclude("/favicon.ico")
                // 鉴权方法：每次访问进入
                .setAuth(obj -> {
                    log.info("come in SaTokenConfigure");
                    // 登录校验 -- 拦截所有路由，并排除/user/doLogin 用于开放登录
                    SaRouter.match("/**", "/sat/admin/login", r -> StpUtil.checkLogin());

                })
                // 异常处理方法：每次setAuth函数出现异常时进入
                .setError(e -> {
                    log.error("error", e);
                    if(e instanceof NotLoginException) {
                        NotLoginException notLoginException = (NotLoginException) e;
                        switch (notLoginException.getLoginType()) {
                            case NotLoginException.NOT_TOKEN:
                                return Resp.fail(RespCodeMsg.UNAUTHORIZED);
                            default:
                                return Resp.fail(RespCodeMsg.INVALID_TOKEN);
                        }
                    }
                    return Resp.fail(RespCodeMsg.SYSTEM_ERROR);
                })
                ;
    }

}
